"""
File   : string_prompt_templates.py
Desc   : 字符串提示模板
Date   : 2024/11/17
Author : Tianyu Chen
"""
from langchain_openai import OpenAI
from langchain_core.prompts import PromptTemplate

llm = OpenAI()

prompt_template = PromptTemplate.from_template("给生产{product}的公司取一个中文名字。")
prompt_value = prompt_template.invoke({"product": "杯子"})


if __name__ == "__main__":
    print(llm.invoke(prompt_value))  # 青玉杯子公司
